import requests, sys
from bs4 import BeautifulSoup


class downloader(object):
    def __init__(self):
        self.server = 'https://www.bqkan.com/'
        self.target = 'https://www.bqkan.com/1_1496/'
        self.names = []
        self.urls = []
        self.nums = 0

    def get_download_url(self):
        req = requests.get(url=self.target)
        req.encoding = req.apparent_encoding
        html = req.text
        div_bf = BeautifulSoup(html, "html.parser")
        div = div_bf.find_all('div', class_='listmain')
        a_bf = BeautifulSoup(str(div[0]), "html.parser")
        a = a_bf.find_all('a')
        self.nums = len(a[12:])
        for each in a[12:]:
            self.names.append(each.string)
            self.urls.append(self.server + each.get('href'))

    def get_contents(self, target):
        req = requests.get(url=target)
        req.encoding = 'gbk'
        html = req.text
        bf = BeautifulSoup(html, "html.parser")
        texts = bf.find_all('div', class_='showtxt')
        texts = texts[0].text.replace('\xa0' * 8, '\n')
        return texts

    def writer(self, name, path, text):
        write_flag = True
        with open(path, 'a', encoding='utf-8') as f:
            f.write(name + '\n')
            f.writelines(text)
            f.write('\n\n')


print("爬取笔趣看小说:https://www.biqooge.com/")
url = input("输入具体书目网址，例：https://www.biqooge.com/1_1496/\n:")
title = input("输入保存文件名:")
if __name__ == '__main__':
    dl = downloader()
    dl.target = url
    dl.get_download_url()
    print("下载:")
    for i in range(dl.nums):
        dl.writer(dl.names[i], 'books\\' + title + '.txt', dl.get_contents(dl.urls[i]))
        sys.stdout.write("已下载:%.3f%%" % float(100 * i / dl.nums) + '\r')
        sys.stdout.flush()
    print('下载完成')
